package cn.acheng.cams.type.utils;


import io.jsonwebtoken.*;
import io.jsonwebtoken.security.Keys;

import javax.crypto.SecretKey;
import java.nio.charset.StandardCharsets;
import java.util.Date;
import java.util.Map;

public class JwtUtil {
    /**
     * 生成jwt
     * 使用Hs256算法, 私匙使用固定秘钥
     *
     * @param secretKey jwt秘钥
     * @param ttlMillis jwt过期时间(毫秒)
     * @param claims    设置的业务信息
     * @return
     */
    public static String createJwt(String secretKey, long ttlMillis, Map<String, Object> claims) {

        // 生成JWT的时间
        long expMillis = System.currentTimeMillis() + ttlMillis;
        Date exp = new Date(expMillis);

        //生成HMAC密钥
        SecretKey key = Keys.hmacShaKeyFor(secretKey.getBytes(StandardCharsets.UTF_8));

        // 设置jwt的令牌生成器
        JwtBuilder builder = Jwts.builder()
                // 设置签名使用的签名算法和签名使用的秘钥，这里没设置签名算法，默认是HS256算法
                .signWith(key)
                // 设置jwt的body
                .claims(claims)
                // 设置过期时间
                .expiration(exp);
        return builder.compact();
    }

    /**
     * Token解密
     *
     * @param secretKey jwt秘钥 此秘钥一定要保留好在服务端, 不能暴露出去
     * @param token     加密后的token
     * @return
     */
    public static Claims parseJWT(String secretKey, String token) {

        //生成HMAC密钥
        SecretKey key = Keys.hmacShaKeyFor(secretKey.getBytes(StandardCharsets.UTF_8));

        // 得到令牌解析器
        JwtParser jwtParser = Jwts.parser()
                // 设置签名的秘钥
                .verifyWith(key)
                .build();
        Jws<Claims> jws = jwtParser.parseSignedClaims(token);
        Claims claims = jws.getPayload();
        return claims;
    }


    /**
     * 获取Token的剩余有效时间（毫秒）
     *
     * @param secretKey jwt密钥
     * @param token 加密后的token
     * @return
     */
    public static long getRemainingTime(String secretKey, String token) {
        try {
            //生成HMAC密钥
            SecretKey key = Keys.hmacShaKeyFor(secretKey.getBytes(StandardCharsets.UTF_8));

            //解析Token
            JwtParser jwtParser = Jwts.parser()
                    .verifyWith(key)
                    .build();
            Jws<Claims> jws = jwtParser.parseSignedClaims(token);
            Claims claims = jws.getPayload();

            //获取过期时间
            Date expiration = claims.getExpiration();

            //计算剩余时间
            long nowMillis = System.currentTimeMillis();
            long expMillis = expiration.getTime();
            return expMillis - nowMillis;
        } catch (JwtException e) {
            //Token无效或解析失败（如签名不匹配、格式错误等）
            return 0;
        } catch (Exception e) {
            //其他异常情况处理
            return 0;
        }
    }
}
